package socket;

import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.List;

/**
 * 服务端
 */
public class Server {
    private ServerSocket serverSocket;
    private List<PrintWriter> allOut = new ArrayList<>();

    public Server() {
        try {
            System.out.println("正在启动服务端。。。");
            serverSocket = new ServerSocket(8088);
            System.out.println("服务端启动完毕！");
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public void start() {
        try {
            while (true) {
                System.out.println("等待客户端连接。。。");
                Socket socket = serverSocket.accept();
                System.out.println("一个客服端连接了！");

                new Thread(new ClientHandler(socket)).start();
            }
        } catch (IOException e) {
            e.printStackTrace();
        }

    }
    public static void main(String[] args) {

        Server server = new Server();
        server.start();

    }
    private class ClientHandler implements Runnable{

        private Socket socket;
        private String host;  // 记录客户端的ip地址信息

        public ClientHandler(Socket socket){
            this.socket = socket;
            host = socket.getInetAddress().getHostAddress();
        }

        @Override
        public void run() {
            PrintWriter pw = null;
            try {
                InputStream ips = socket.getInputStream();
                InputStreamReader isr = new InputStreamReader(ips, StandardCharsets.UTF_8);
                BufferedReader br = new BufferedReader(isr);

                OutputStream ops = socket.getOutputStream();
                pw = new PrintWriter(new BufferedWriter(
                                 new OutputStreamWriter(ops, StandardCharsets.UTF_8)),true);
                allOut.add(pw);
                System.out.println(host + "上线了，当前在线人数：" +allOut.size());


                String message;
                while ((message = br.readLine()) != null) {
                    System.out.println(host+"说："+message);
                    for (PrintWriter o:allOut) {
                        o.println(host+"说："+message);
                    }
                }
            } catch (IOException e) {
                System.out.println("客户端"+host+"断开连接");
            }finally {
                try {
                    allOut.remove(pw);
                    System.out.println(host+"下线了，当前人数："+allOut.size());
                    socket.close();
                } catch (IOException e) {
                    System.out.println("客户端断开连接");
                }
            }
        }
    }
}
